/**
 * Software License, Version 1.0
 * 
 * Copyright 2003 The Trustees of Indiana University.  All rights reserved.
 * 
 *
 *Redistribution and use in source and binary forms, with or without 
 *modification, are permitted provided that the following conditions are met:
 *
 *1) All redistributions of source code must retain the above copyright notice,
 * the list of authors in the original source code, this list of conditions and
 * the disclaimer listed in this license;
 *2) All redistributions in binary form must reproduce the above copyright 
 * notice, this list of conditions and the disclaimer listed in this license in
 * the documentation and/or other materials provided with the distribution;
 *3) Any documentation included with all redistributions must include the 
 * following acknowledgement:
 *
 *"This product includes software developed by the Community Grids Lab. For 
 * further information contact the Community Grids Lab at 
 * http://communitygrids.iu.edu/."
 *
 * Alternatively, this acknowledgement may appear in the software itself, and 
 * wherever such third-party acknowledgments normally appear.
 * 
 *4) The name Indiana University or Community Grids Lab or NaradaBrokering, 
 * shall not be used to endorse or promote products derived from this software 
 * without prior written permission from Indiana University.  For written 
 * permission, please contact the Advanced Research and Technology Institute 
 * ("ARTI") at 351 West 10th Street, Indianapolis, Indiana 46202.
 *5) Products derived from this software may not be called NaradaBrokering, 
 * nor may Indiana University or Community Grids Lab or NaradaBrokering appear
 * in their name, without prior written permission of ARTI.
 * 
 *
 * Indiana University provides no reassurances that the source code provided 
 * does not infringe the patent or any other intellectual property rights of 
 * any other entity.  Indiana University disclaims any liability to any 
 * recipient for claims brought by any other entity based on infringement of 
 * intellectual property rights or otherwise.  
 *
 *LICENSEE UNDERSTANDS THAT SOFTWARE IS PROVIDED "AS IS" FOR WHICH NO 
 *WARRANTIES AS TO CAPABILITIES OR ACCURACY ARE MADE. INDIANA UNIVERSITY GIVES
 *NO WARRANTIES AND MAKES NO REPRESENTATION THAT SOFTWARE IS FREE OF 
 *INFRINGEMENT OF THIRD PARTY PATENT, COPYRIGHT, OR OTHER PROPRIETARY RIGHTS. 
 *INDIANA UNIVERSITY MAKES NO WARRANTIES THAT SOFTWARE IS FREE FROM "BUGS", 
 *"VIRUSES", "TROJAN HORSES", "TRAP DOORS", "WORMS", OR OTHER HARMFUL CODE.  
 *LICENSEE ASSUMES THE ENTIRE RISK AS TO THE PERFORMANCE OF SOFTWARE AND/OR 
 *ASSOCIATED MATERIALS, AND TO THE PERFORMANCE AND VALIDITY OF INFORMATION 
 *GENERATED USING SOFTWARE.
 */
package cgl.narada.discovery.broker.bdn;

import java.io.DataInputStream;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.Hashtable;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.log4j.Logger;

import cgl.narada.discovery.broker.messages.BrokerAdvertisement;
import cgl.narada.util.ObjectMarshaller;
import cgl.narada.util.UUIDRetriever;

/**
 * <br>
 * Created on Mar 28, 2005
 * 
 * @author Harshawardhan Gadgil (hgadgil@grids.ucs.indiana.edu)
 */
public class BDN extends HttpServlet {
    static Logger log = Logger.getLogger("BDN");

    public void doPost(HttpServletRequest request, HttpServletResponse res)
            throws IOException, ServletException {

        DataInputStream byteStream = new DataInputStream(request
                .getInputStream());
        byte[] brokerAdBytes = new byte[request.getContentLength()];

        byteStream.readFully(brokerAdBytes);

        BrokerAdvertisement bad = null;

        log.info("Recieved " + brokerAdBytes.length + " Bytes");
        try {
            bad = (BrokerAdvertisement) ObjectMarshaller
                    .unmarshall(brokerAdBytes);
        } catch (IOException e) {
            e.printStackTrace();
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }

        log.info("Host: " + bad.getHostname());
        log.info("Info: " + bad.getInfo());
        log.info("Network: "  + bad.getBelongsToNetwork());
        Hashtable supportedProtocols = bad.getSupportedTransportProtocols();

        log.info(supportedProtocols.size() + " Supported Protocols");

        String brokerUUID = (UUIDRetriever.getInstance())
                .getTimeBasedUUIDAsString();

        BrokerTableEntry brokerTableEntry = new BrokerTableEntry();
        brokerTableEntry.brokerUUID = brokerUUID;
        brokerTableEntry.brokerAd = bad;
        log.info("Setting Request UUID: " + brokerUUID);
        brokerTableEntry.brokerConnection = new BrokerConnection(brokerUUID, bad);
        BrokerTableManager.put(brokerUUID, brokerTableEntry);
        
        PrintWriter out = new PrintWriter(res.getOutputStream());

        out.write("Registered Broker! UUID:" + brokerUUID);
        out.flush();
    }

    public void doGet(HttpServletRequest request, HttpServletResponse response)
            throws IOException, ServletException {
        response.setContentType("text/html");
        PrintWriter out = response.getWriter();
        out.println("<h1>Broker Discovery Node Registration Servlet</h1>");
        out.println("<b>Only POST requests allowed !</b>");
    }

}
